import unittest
import ChessBoard
from ChessConsts import ChessConsts

class testChessBoard(unittest.TestCase):
    def testResetBoardResetsMove(self):
        sut = ChessBoard.ChessBoard()
        sut.addTextMove('b2 b4')
        sut.resetBoard()
        self.assertEquals(sut._board[6][1], 'P')

    def testResetBoardResetsColour(self):
        sut = ChessBoard.ChessBoard()
        sut.addTextMove('e2 e4')
        sut.resetBoard()
        self.assertEquals(sut._turn, ChessConsts.WHITE)

if __name__ == '__main__':
    unittest.main()
